from 卷积神经网络 import predict_image

PATH_best = 'model/image_classification_best.pth'
print(predict_image('my_data/0.png', PATH_best))
print(predict_image('my_data/1.png', PATH_best))
print(predict_image('my_data/2.png', PATH_best))
print(predict_image('my_data/3.png', PATH_best))
print(predict_image('my_data/4.png', PATH_best))
print(predict_image('my_data/5.png', PATH_best))
print(predict_image('my_data/6.png', PATH_best))
print(predict_image('my_data/7.png', PATH_best))
print(predict_image('my_data/8.png', PATH_best))
print(predict_image('my_data/9.png', PATH_best))